package com.lagou.utils;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;

/*
*   连接工具类：从数据源中获取一个连接，并且将获取到的连接与线程进行绑定
*   ThreadLocal: 线程内部的存储类，可以在指定的线程内存储数据 key:threadlocal(当前线程) value:任意类型的值
*/
@Component
public class ConnectionUtils {
    @Autowired
    private DataSource dataSource;

    private ThreadLocal<Connection> threadLocal = new ThreadLocal<>();
//  获取当前线程上绑定的连接：如果获取到的连接为空，那么就要从数据源中获取连接，并放到ThreadLocal中（绑定到当前线程）
    public Connection getThreadConnection(){
//        1、先从ThreadLocal上获取连接
        Connection connection = threadLocal.get();
//        2、判断当前线程中是否有Connection
        if(connection == null){
            // 3、从数据源中获取一个连接，并且存入ThreadLocal中
            try {
                connection = dataSource.getConnection();
            } catch (SQLException throwables) {
                throwables.printStackTrace();
            }
            threadLocal.set(connection);
        }
        return connection;
    }

    /*
        解除当前线程的连接绑定
    */
    public void removeThreadConnection(){
        threadLocal.remove();
    }
}
